import torch


def save_model(_model, _optimizer, epoch, path):
    state = {'net': _model.state_dict(), 'optimizer': _optimizer.state_dict(), 'epoch': epoch}
    torch.save(state, path)

